package com.tanhua.recommend.listener;

import cn.hutool.core.util.StrUtil;
import com.tanhua.model.mongo.Movement;
import com.tanhua.model.mongo.MovementScore;
import com.tanhua.model.mongo.Video;
import com.tanhua.model.mongo.VideoScore;
import lombok.extern.slf4j.Slf4j;
import org.bson.types.ObjectId;
import org.springframework.amqp.core.ExchangeTypes;
import org.springframework.amqp.rabbit.annotation.Exchange;
import org.springframework.amqp.rabbit.annotation.Queue;
import org.springframework.amqp.rabbit.annotation.QueueBinding;
import org.springframework.amqp.rabbit.annotation.RabbitListener;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.stereotype.Component;

import java.util.Map;

@Component
@Slf4j
public class RecommendVideoListener {

    @Autowired
    private MongoTemplate mongoTemplate;

    @RabbitListener(bindings = @QueueBinding(value = @Queue(value = "tanhua.video.queue"),
            exchange = @Exchange(value = "tanhua.log.exchange", type = ExchangeTypes.TOPIC),
            key = {"log.video"}))
    public void recommend(Map<String, Object> paramMap) {
        log.info("=======================开始处理消息数据=======================");
        Long userId = Long.valueOf(paramMap.get("userId").toString());
        String type = (String) paramMap.get("type");
        String logTime = (String) paramMap.get("logTime");
        String objId = paramMap.get("objId").toString();
        Video video = mongoTemplate.findById(new ObjectId(objId), Video.class);
        Double score = getScore(type);

        if (video != null) {
            VideoScore videoScore = new VideoScore();
            videoScore.setUserId(userId);
            videoScore.setVideoId(video.getVid());
            videoScore.setScore(score);
            videoScore.setDate(System.currentTimeMillis());
            mongoTemplate.save(videoScore);
        }

    }

    private static Double getScore(String type) {
        //0301为发小视频，0302为小视频点赞，0303为小视频取消点赞，0304为小视频评论
        Double score = 0d;
        switch (type) {
            case "0301":
                score = 2d;
                break;
            case "0302":
                score = 5d;
                break;
            case "0303":
                score = -5d;
                break;
            case "0304":
                score = 10d;
                break;
            default:
                break;
        }
        return score;
    }
}
